package com.example.util;

import com.alibaba.fastjson.JSON;

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class ShortMsgUtil {
    private final static String DEF_CHATSET = "UTF-8";
    private final static int DEF_CONN_TIMEOUT = 30000;
    private final static int DEF_READ_TIMEOUT = 30000;
    private final static String userAgent = "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/29.0.1547.66 Safari/537.36";
    private final static String url = "http://v.juhe.cn/sms/send";

    private static String appKey;

    public static ShortMsgUtil init(String key) {
        appKey = key;
        return new ShortMsgUtil();
    }

    public static class Result {

        private final static Integer SUCCESS = 0;
        private final static Integer FREQUENT_SEND = 1005;


        private Integer error_code;
        private String reason;

        public boolean isFrequentSend() {
            return FREQUENT_SEND.equals(error_code);
        }

        public boolean isSuccess() {
            return SUCCESS.equals(error_code);
        }

        public Integer getError_code() {
            return error_code;
        }

        public void setError_code(Integer error_code) {
            this.error_code = error_code;
        }

        public String getReason() {
            return reason;
        }

        public void setReason(String reason) {
            this.reason = reason;
        }

    }


//    /**
//     * @param phone    接收短信的手机号码
//     * @param tplId    短信模板ID，请参考个人中心短信模板设置
//     * @param tplValue #code#=1234&#company#=聚合数据
//     * @return
//     */
//    public Result sendMsg(String phone, Integer tplId, String tplValue) {
//        String formatUrl = String.format(url, phone, tplId, tplValue, appKey, "json");
//        String resultStr = HttpKit.get(formatUrl);
//        Result result = JSON.parseObject(resultStr, Result.class);
//        return result;
//    }

    /**
     * @param phone    接收短信的手机号码
     * @param tplId    短信模板ID，请参考个人中心短信模板设置
     * @param tplValue #code#=1234&#company#=聚合数据
     * @return
     */
    public Result sendMsg(String phone, Integer tplId, String tplValue) throws Exception {
        Map<String, String> paras = new HashMap<>();
        paras.put("mobile", phone);
        paras.put("tpl_id", tplId + "");
        paras.put("tpl_value", tplValue);
        paras.put("key", appKey);
        paras.put("dtype", "json");
        String resultStr = net(url, paras, "GET");
        Result result = JSON.parseObject(resultStr, Result.class);
        return result;
    }


    /**
     * @param phone
     * @param tplId
     * @param tplValues
     * @return
     * @throws Exception
     */
    public Result sendMsg(String phone, Integer tplId, Map<String, String> tplValues) throws Exception {
        List<String> variableStrings = new ArrayList<>();
        for (String key : tplValues.keySet()) {
            variableStrings.add(String.format("#%s#=%s", key, tplValues.get(key)));
        }
        return sendMsg(phone, tplId, String.join("&", variableStrings));
    }

    /**
     * @param strUrl 请求地址
     * @param params 请求参数
     * @param method 请求方法
     * @return 网络请求字符串
     * @throws Exception
     */
    public static String net(String strUrl, Map<String, String> params, String method) throws Exception {
        HttpURLConnection conn = null;
        BufferedReader reader = null;
        String rs = null;
        try {
            StringBuffer sb = new StringBuffer();
            if (method == null || method.equals("GET")) {
                strUrl = strUrl + "?" + urlencode(params);
            }
            URL url = new URL(strUrl);
            conn = (HttpURLConnection) url.openConnection();
            if (method == null || method.equals("GET")) {
                conn.setRequestMethod("GET");
            } else {
                conn.setRequestMethod("POST");
                conn.setDoOutput(true);
            }
            conn.setRequestProperty("User-agent", userAgent);
            conn.setUseCaches(false);
            conn.setConnectTimeout(DEF_CONN_TIMEOUT);
            conn.setReadTimeout(DEF_READ_TIMEOUT);
            conn.setInstanceFollowRedirects(false);
            conn.connect();
            if (params != null && method.equals("POST")) {
                try {
                    DataOutputStream out = new DataOutputStream(conn.getOutputStream());
                    out.writeBytes(urlencode(params));
                } catch (Exception e) {
                    // TODO: handle exception
                }
            }
            InputStream is = conn.getInputStream();
            reader = new BufferedReader(new InputStreamReader(is, DEF_CHATSET));
            String strRead = null;
            while ((strRead = reader.readLine()) != null) {
                sb.append(strRead);
            }
            rs = sb.toString();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (reader != null) {
                reader.close();
            }
            if (conn != null) {
                conn.disconnect();
            }
        }
        return rs;
    }

    //将map型转为请求参数型
    private static String urlencode(Map<String, String> data) {
        StringBuilder sb = new StringBuilder();
        for (Map.Entry<String, String> i : data.entrySet()) {
            try {
                sb.append(i.getKey()).append("=").append(URLEncoder.encode(i.getValue() + "", "UTF-8")).append("&");
            } catch (UnsupportedEncodingException e) {
                e.printStackTrace();
            }
        }
        return sb.toString();
    }
}
